package com.okaixz.client.model.vo;

import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;

/**
 * @author 陈大专
 * @time 2024/7/23 8:43
 * @des
 **/
@NoArgsConstructor
@Data
public class AiResult {

    private Integer code;
    private String message;
    private String sid;
    private List<ChoicesDTO> choices;
    private UsageDTO usage;

    @NoArgsConstructor
    @Data
    public static class UsageDTO {
        private Integer prompt_tokens;
        private Integer completion_tokens;
        private Integer total_tokens;
    }

    @NoArgsConstructor
    @Data
    public static class ChoicesDTO {
        private MessageDTO message;
        private Integer index;

        @NoArgsConstructor
        @Data
        public static class MessageDTO {
            private String role;
            private String content;
        }
    }
}
